echo "preprocess"

python3 export_onnx.py

onnxsim onnx/clip.onnx onnx/clip_opt.onnx
onnxsim onnx/hint.onnx onnx/hint_opt.onnx
onnxsim onnx/controlnet.onnx onnx/controlnet_opt.onnx
onnxsim onnx/decoder.onnx onnx/decoder_opt.onnx

python3 clip_surgeon.py
python3 controlnet_surgeon.py
python3 decoder_surgeon.py

trtexec --onnx=clip_opt.onnx --fp16 --saveEngine=clip_fp16.trt --workspace=22000 --verbose --useSpinWait --plugins=./libplugin.so
trtexec --onnx=onnx/hint_opt.onnx --fp16 --saveEngine=hint_fp16.trt --workspace=22000 --verbose --useSpinWait
trtexec --onnx=controlnet_opt.onnx --fp16 --saveEngine=controlnet_fp16.trt --workspace=22000 --verbose --useSpinWait --plugins=./libplugin.so
trtexec --onnx=decoder_opt.onnx --fp16 --saveEngine=decoder_fp16.trt --workspace=22000 --verbose --useSpinWait --plugins=./libplugin.so
#python3 build_decoder.py